Added the win state, and got it to work