Add figures code