added some plotting to try and figure out results