Add prime generator
Add problem 3