Bug fixing: test was using wrong dataset data and doc changes