Add type hints
Add type hints to screenshot! function