Add the new classes (error from last commit)