Add if __name__ = __main__