add colour coding for display