Added support for ui-router