Change AngularJS controller to accept multiple state