Updated code styling and added braces.