Minor modification (added braces)