cleaning up script, adding check for ubuntu version