add test set for controller
add model functionality