Update ensetup script