Fiddled around with the plotting commands a bit