Add tests
Create documentaion