Add setup to script