Added UIModel class