added some more plotting