Updated references and added figure to text