update the script in to test on ubuntu