Add citation and github repo link