Add citation for code