forgot to add new classes