Write selenium code