Added figure class