added screenshot calls

Added screenshot code.