Tech Unicorn interview question

Write a function to check prime numbers