Forgot to add new class to last commit