def setup_lines(ax, Yshape, labs, plott):
    """Sets up artist for animation or solution snaps

    Parameters
    ----------
    ax :
        axes to use for artist
    Yshape : tuple
        shape of the solution array
    labs : list
        labels for the solution
    plott : str ("steps" or "plot")
        type of plot to use

    Returns
    -------

    lines
    """
    if plott is None:
        plott = ax.plot
    else:
        plott = ax.__getattribute__(plott)

    if len(Yshape) > 2:
        lines = [plott([], [], lw=2)[0] for foo in range(Yshape[2])]
        for i, l in enumerate(lines):
            if labs is None:
                l.set_label("q" + str(i + 1) + "(x, t)")
            else:
                l.set_label(labs[i])
    else:
        lines, = plott([], [], lw=2)
        if labs is None:
            lines.set_label("q(x, t)")
        else:
            lines.set_label(labs)
    return lines