Update install script

install.sh

@@ -6,10 +6,20 @@ if [ $# -eq 0 ]
     exit 1;
 fi
 
+BRANCH=$1
+
+
 echo ".dotfiles" >> $HOME/.gitignore
+
 git clone --bare https://github.com/LordMathis/dotfiles.git $HOME/.dotfiles
-alias config='/usr/bin/git --git-dir=$HOME/.dotfiles/ --work-tree=$HOME'
-config checkout $1
+
+function config {
+   /usr/bin/git --git-dir=$HOME/.dotfiles/ --work-tree=$HOME "$@"
+}
+
+cd $HOME
+
+config checkout $BRANCH
+
 config submodule init
-config submodule update
-source .zshrc
+config submodule update --recursive --remote