Refactored colour setting of arrows into own method