Add function drawGradient