Small code style changes

I made some small code style tweaks.