added dataset and small changes made