added prime_gen, changed unit_tests