Add data model
Close #3