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