Option for screenshots

Added an option for screenshots to be taken on failures/timeouts.