made code better