Add references and fix up related works