Change to figure.set_title