Rewrite setup.sh to handle .vim folder