add more colour to prompt

With thanks to @brianm https://gist.github.com/brianm/7684710