Updated docs to reflect recent refactoring