Update puzzle state representation to include 'expansion order'.