Adding screenshot taking functionality.