Add support for ui-router