Made some minor PEP8 related changes

Signed-off-by: Benjamin Ruston <benjy.ruston@gmail.com>