Diquark confinement in an extended NJL model