Added ability to register account. Small fix