Yeah.. forgot to include the new class..