added some fancy plotting