Update tests, documentation