Improve the following code | PromptsAI