minor fixes in plotting