Got replay functionality working for basic case