Removed pickle import that I left behind during some testing.