Wrote code to save useful data in pickle file