Write a python code to calculate prime numbers