added some styling improvements and better prompts