Make sure I include the new class into the gem