added braces per review