Updated to use the new screenshot method.