Add recording functionality