modified some code, updated figures, added figure