Linted up the code!