Updated PlayState to work with LoseState