Add model and modify controller

Added model and modified the controller to use the correct entity
framework model