Adding to dataset model. Fixes #23.