def elite_selection(population, elite_num):
    """
    Uses elite selection to pick elite_num units from population

    :param population: population to be picked from
    :param elite_num: number of units to be selected
    :return: array of selected units
    """
    return population[:elite_num]