adding some code for practise
few changes adding figure