COLA with scale-dependent growth: applications to screened modified gravity models