Added code to perform as expected
Add comments