Updated JS slightly