document code better