Added ability to register