wrap this in if __name__ == '__main__'