Make registering feature more flexibile.