ZuciTech Interview Question

Write a Python function to check whether the number is prime or not