def METHOD_NAME(class_id):
    """
    Utility function used to get the corresponding class from a class ID.
    :param class_id: String representing the class ID
    :return: Challenge class
    """
    cls = CHALLENGE_CLASSES.get(class_id)
    if cls is None:
        raise KeyError
    return cls