Better API in model key