Add storage function

Not fully functional