Removed if __name__ =='__main__'