added more model features