Added colour switch to im array script