added model but does not work yet