Add comment suggesting pickle for storing data