Modified code for bharat's data set