Prime class and a test