Update main

added a function for finding prime numbers