Replaced collapse with ng-if