Updated UseCase