Add onPreview hook