Improve model comments