updated figures and code