Add screen shots from test results after code improvements