Fix pep8 style issues

Signed-off-by: Joe Stringer <joestringernz@gmail.com>