Add prime class
Add a simple checker