Updated the code documentation

Updated the code documentation