Enter 5 numbers in two sets and find the difference update of the first set in Python

Sets - Question 5

Q5) Write a program in Python to enter 5 numbers in two sets and find the difference update of the first set, that means update Set A with numbers which are present only in Set A but not in Set B and display Set A on the screen.

    Example:-
    Set A = {1, 2, 3, 4, 5}
    Set B = {1, 6, 2, 7, 3}
    Set a after difference update
    {4, 5}

a=set()
b=set()

print('Enter 5 numbers in Set a')
for i in range(5):
    a.add(int(input()))

print('Enter 5 numbers in Set b')
for i in range(5):
    b.add(int(input()))

# Update Set a having numbers only present in Set a but not in Set b
a.difference_update(b)
print('Set a after difference update')
print(a)

Enter 5 numbers in Set a
1
2
3
4
5
Enter 5 numbers in Set b
1
6
2
7
3
Set a after difference update
{4, 5}