Made some code a bit nicer. jk-5, watch your code style!
small fix