added figures for writeup