Add code for figure

Update bibliography