modifying prime number code 1.1