Add tests for ifs