Some style

Made some small style changes while getting a look at the code.