Fixed another bug 

I have fixed another bug and added some colour to the text and if your not sure about anything please ask me as you reintroduced a bug with your last commit that I have just fixed