Added KeyModel class