Simple change in plotting