making changes to the PR as requested, I am including data sets for testing purposes.