Made dataset initializers more generic-looking.