Clean up code
(Thanks IntelliJ)