Move screenshot logic to scripts