Added functionality for if statements.