Improve the following code | YutoAI