Changed hwere the mosaic plot warning occurs in code