Minor error in the code fixed