Added some figures, and code