Beautified the code just a tiny bit!