I want to improve your code.