Add "if __name__ == '__main__'"