Added classes for the database with relationships