Updated documentation 

Updated to to reflect new code implementation