Forgot a class in last commit....