1) Used TreeMap to retain the natural ordering of keys in hashmap. 2) Centralised all the meta information about datasets in the Dataset class. 3) Added support for testing the REFERENCE TESTDATASET mentioned in the paper.