fix the sizing issue with the plots