Adding prompt
Should have put that in main
Don't need to specify test