Add ability to end replay mode