add some model for better database support