add data storage class
Add some UI implementation