Adding storage schema and updating UI