Refactor winner code to model