Added pickle for serialization