Setup script updated