Add ability to register