Former Chelsea manager Roberto Di Matteo thinks Chelsea should try and keep Diego Costa, but only if the striker is happy at Stamford Bridge.