added predict fucntion from Dataset