Allow braceless ifs and whiles

Closes #2