Added import for datasets.
Fixed bug in putting images.
Merge branch 'test' into dev