Replaced the git branch prompt func with a better one.