Refactor ng-if, more proper like this :)