added more graphs and testing for prediction