Add replay feature