clean up win/lose logic