Add IMDB dataset helper function