Improve the following code | Prompt Hub