Add support for controllerAs