Update zshrc to run on ubuntu too