Added replay feature. Still need to be able to close replay box