Add function to save YUV frames