Add code documentation