if choice == 1:
months = int(input("enter number of months: "))
if emp_id in sup_id:
tot_salary = sup.salary(months)
print(tot_salary)
else:
tot_salary = emp.salary(months)
print(tot_salary)
elif choice == 2:
if emp_id in sup_id:
sup.disp_details()
else:
emp.disp_details()
elif choice == 3:
if emp_id in sup_id:
sup.check()
else:
emp.check()
elif choice == 4:
if emp_id in sup_id:
sup.calc_tax()
else:
emp.calc_tax()
else:
break
