minor changes
updated model