updating documentation and fixing a test