added some plotting