Fix code documentation