Ok now it is working
Added more formatting to the figure