Adding more angularjs logic