Stop writing long Selenium Python code