Add tests, gems etc...
Finish regestration functionality