Few minor style fixes