add couple of ng-ifs