Added support for if statements