Add ability to capture snapshot