Improve the following code | 3H33 Prompt Lab