Improve in-code documentation