Add a win & lose logic