Spain international Diego Costa is set to remain at Atletico Madrid until the summer of 2021 despite his poor form since leaving Chelsea back in 2017