Add screenshot removal code