bash/rvm-install in rvm-0.0.4 vs bash/rvm-install in rvm-0.0.5

@@ -1,26 +1,31 @@
 #!/bin/bash
 
 mkdir -p ~/.rvm/src ~/.rvm/bin
 
-cwd=`pwd`
-cp $cwd/bash/rvm ~/.rvm/bin/
+cp $(pwd)/bash/rvm ~/.rvm/bin/
 
-touch ~/.bash_profile
-
-if [ -z "`grep '\.rvm/bin/rvm' ~/.bash_profile`" ] ; then
-  echo "Adding 'source ~/.rvm' to your ~/.bash_profile."
-  echo "source ~/.rvm/bin/rvm" >> ~/.bash_profile
+touch ~/.profile
+if [ -z "`grep '\.rvm/bin/rvm' ~/.profile`" ] ; then
+  echo "Adding 'source ~/.rvm/bin/rvm' to your ~/.profile."
+  echo "source ~/.rvm/bin/rvm" >> ~/.profile
 fi
 
-if [ -z "`grep '\.rvm/current' ~/.bash_profile`" ] ; then
-  echo "Adding 'source ~/.rvm/current' ~/.bash_profile."
-  echo "if [ -f ~/.rvm/current ] ; then source ~/.rvm/current ; fi" >> ~/.bash_profile
+if [ -z "`grep '\.rvm/current' ~/.profile`" ] ; then
+  echo "Adding 'source ~/.rvm/current' ~/.profile."
+  echo "if [ -f ~/.rvm/current ] ; then source ~/.rvm/current ; fi" >> ~/.profile
 fi
 
+touch ~/.bash_profile
 if [ -z "`grep '\.profile' ~/.bash_profile`" ] ; then
   echo "Adding 'source ~/.profile' to the end of ~/.bash_profile."
   echo "if [ -f ~/.profile ] ; then source ~/.profile ; fi" >> ~/.bash_profile
 fi
 
-source ~/.rvm/bin/rvm
+touch ~/.zprofile
+if [ -z "`grep '\.profile' ~/.zprofile`" ] ; then
+  echo "Adding 'source ~/.profile' to the end of ~/.zprofile."
+  echo "if [ -f ~/.profile ] ; then source ~/.profile ; fi" >> ~/.zprofile
+fi
+
+echo -e "\n\e[0;32mOpen a new shell and enjoy rvm: rvm usage\e[0m\n"