Update unit tests
Update documentation