add class for db