'Beautified' the code, meh