Напишите функцию для вычисления единичного вектора

Домашнее задание записывается следующим образом:

Напишите функцию с именем unitVec, которая определяет единичный вектор в направлении линии, соединяющей две точки (A и B) в пространстве. Функция должна принимать на вход два вектора (списка), каждый с координатами точки в пространстве. На выходе должен быть вектор (список) с компонентами единичного вектора в направлении от A к B. Если точки A и B имеют по две координаты каждая (т. е. лежат в плоскости xy), то выходной вектор должен иметь две элементы. Если точки A и B имеют по три координаты каждая (т. е. лежат в общем пространстве), выходной вектор должен состоять из трех элементов.

У меня в основном написан весь код, но я не могу понять, как возвести в квадрат каждый элемент в списке с именем connects[].

Чтобы вычислить единичный вектор, программа вычтет элементы вектора B из соответствующих элементов вектора A и создаст новый список (connects[]) с этими значениями. Затем каждый из этих элементов нужно возвести в квадрат и все их нужно сложить вместе. Затем из этого числа будет взят квадратный корень, и каждый элемент в connects[] будет разделен на это число и сохранен в новом списке, который будет единичным вектором. Я пытаюсь добавить квадраты элементов в connects [], используя строку

add = add + (connects[i]**2)

но я знаю, что это возвращает список только дважды. Остальная часть моего кода в порядке, мне просто нужна помощь в возведении этих элементов в квадрат.

from math import *

vecA = []
vecB = []
unitV = []
connects = []
vec = []

elements = int(input("How many elements will your vectors contain?"))


for i in range(0,elements):
    A = float(input("Enter element for vector A:"))
    vecA.append(A)
    B = float(input("Enter element for vector B:"))
    vecB.append(B)  

def unitVec(vecA,vecB):
    for i in range(0,elements):
        unit = 0
        add = 0
        connect = vecB[i] - vecA[i]
        connects.append(connect)
        add = add + (connects[i]**2)
        uVec = sqrt(add)
        result = connects[i]/uVec
        unitV.append(result)
    return unitV

print("The unit vector connecting your two vectors is:",unitVec(vecA,vecB))