Making screenshot part of task flow