fix to arrangeGrob plot and return value