Updated the class name, forgot to do that previously