Add support for saving screenshots