Work on setup script