Replace $SHELL method to detect bash shell by $BASH and $ZSH_NAME (for bash and zsh respectively)