Dataset utility functions (#49)

* Made simple agents suitable for challenge submissions.

* Update docker

* Addressed the comments, chaned flag names, switched to benchmark

* Removed .swp file, added swp to gitignore.

* Adding arxiv identifier

* dataset changes

* added tests, changed interface slightly

* added collate parameter

* addressed oleksandr's comments