Making a new class for testing this function sounds like a good idea