i made your code better