added data modeling for database