Made code better formatted