Made the look of the code better