fix header and documentation
more tests