Setup script update