Added input and img elements
move the test.py file to be less generic