Write a python program using return type with function to input a number and check if the number is prime or composite number using nested if-else loop.