End of day commit --- working on plotting prediction data