Added if __name__ == "__main__"