Slight style fixes