Class 158 & 170 Diesel Multiple Units are set for refurbishment