Prime number passing an argument to main function