Balance models better and rewrite caption