formatting the code better