Diego Costa hace el segundo tras una buena recuperación