Flake8 fixes

Signed-off-by: Vasudev Kamath <kamathvasudev@gmail.com>