How to structure a model to plot multiple entities?

Try this gist as an example: estimate-means-missing-gaussian · GitHub. You are almost right with the model you had. It was just missing some shape info, I think.