Improved the prompt a bit more