Set colour correctly