Adding the draw callback