Oops forgot a class