Add more functions to check prime