add test for if/else