Adding figures / explanation of the code