removing if __name__ == '__main__'