Added some minor style fixes