Add screenshot functionality.