added colour and some AI improvement