Add dataset exploration script