Diego Costa, renovado con el Atleti