Add in legend to AI post.