Bugfixes and refactoring for prediction code, added some TODO notes.
Add LD29 dataset