Add prime calculation
Change to long to allow for problem 3 to run