Put some scripts behind __name__ == '__main__'