add the partial observation part