Added code to perform training.
Started debugging.
Added some hacks into the Replay dataset to handle undocument requirements of a dataset.