whoops - adding figure back in