update to class exercise