Create data_storage.py