Added tests
Update docs