oops forgot to include the new class