Updated model and controller to match change on database.