Refactor mock classes