Improved rainbow colouring