Updated Python Automation to include screenshot