Added TODO to fix performance of prime generator