yam = input("Enter num: ") if yam.isdigit(): yam = print(pow(int(yam),2)) else: yam = input("Enter num: ")