Added capability to capture screens at end of generation.