Add win/lose checking

(TODO: win/lose screens)