Updated code and figures.