removed 'if __name__ == __main__:    '