Refactor how dataset attributes are set.
Added more documentation for the class.