added figures, updated demo
changed something apparently