Migrate to ui-router