def draw_text(ax, x, y, text, fontsize, color='black'):
    """Wrapper for drawing text"""
    ax.text(x, y, text, fontsize=fontsize, color=color, horizontalalignment
        ='center', verticalalignment='center')
