pep8 cleanup

@jzoldak here you go