Updated part of the code documentation