Made the code look pretty