Small mod of zsh theme