Add if, return, and variable assignment statements