added some plotting capabilities