Update documentation to reflect interface change
Fixing documentation to reflect changes