Update tests and documentation