added if __name == '__main__', made PEP8 compliant