def codehilite_extension(line_numbers, css_code_class):
    """
    Returns the extension with correct config for the custom codehilite.
    """
    codehilite = [
        "codehilite(css_class=",
        css_code_class,
        ", linenums=",
        str(line_numbers),
        ")"
        ]
    return "".join(codehilite)