Updated README and figures