Add data storage functionality