Adding code to read data back in from disk