Add solution for data models exercise