Nit: please use braces even for single line statements.