improved test
improved docs