What should I change on my code?