Add tests for IF THEN ELSE