Updated mock classes