small changes + plot