Add better prompts