Add more context to prompt