Added a figure
Updated methods