Update to do in class