Added Ubuntu support to main script