add screenshot feature.

$screenshot = $driver->takeScreenshot($save_as);

$driver->takeScreenshot() return the png image.
If you want to save the screenshot to somewhere, pass the path of as
$save_as. If $save_as is null, then it won't be save.