Refactor interfaces of the FigureWidget