Customized the prompt a bit more