Take a screenshot after every step