Just style fixes