Add legend + save image works better