Importing pickle module for future use and minor clean up