cleaning up and adding some example figures