Silly question, of course I should, but where? I checked the file it was copied from and it is in the main import section at the top of the file, but it isn't needed unless the file is run directly, so should it be in the `if __name__ == '__main__':` section?