Tidied up the __main__ prompts