Included forgotten new class in last commit