Could you add this class to the Tests folder?