Modified JS for party select