how can i improve my code?