updated docs and refactoring to correct graphs