small bug fixes and plot improvements.