Updated in-code documentation