Add two class for the DB