Add stop watch, more state to AI