It was being split into two code blocks at the ```if __name__ == '__main__':```