make the code better