improve key model and code