add citation and address more reviewer comments