Made the code prettier