Removed the pickle import