model improvments