change for pep8 and add def, __name__ == __main__