Added ability to take screenshots