Added a figure to test