Data model changes (v2) 

Original PR: #51 


Yeah I agree with you.