flake: Fix flake's formatting issues

Flake aims to provide style guide enforcement,
this commit applies those style guidelines whenever
missing in our code.

Signed-off-by: Ahmed Ismail <Ahmed.Ismail@arm.com>