Updating documentation after refactor