Added code for replaying of generated dataset