Get if/else working