Minor bug fix

Fixed some small things.
Small style fix

Gonna learn this shit lol