Fix image removal after test
Add ability to capture specific DOM elements

Fixes #100.