[wallace] fixing the screenshot to only take if the scenario fails. Removing unneeded printlines and commented code. Also, correct a switch statement to an if. Switch cant be used with strings.