flake8 corrections

Signed-off-by: David Saradini <david.saradini@me.com>