Fix column check and column colouring