added model for db