added testbench and screenshot code now working