Update script to work on Ubuntu machines.