Updated README file to include working code for Figure 2