Few small style fixes