added colour and a little tweaking