Add setup script to yml