Use case instead of if