forgot to add new class in last commit