Added pickle for saved data
Added test plots