added model for database