just a little polish

add braces for if and for expressions