Update documentation + tests