PART - 2 code added