Style changes per PR feedback