Added plotting for REML