Adding support for preview button.