Add back code documentation