Improve zsh and irb/pry functions