Replay functionality added