What's this `input()` for? I think we should avoid waiting for the user without giving some kind of prompt. You could do `time.sleep(1)`, or `input("press enter")`.