cs1 task4 - datamodel draft