On save, set container height to text height