Made the code prettier :)