Added a figure for the UI