Modify script to support URL column