def norm_dir_vect(A, B):
    """
    Normalized direction vector from point A to point B

    Parameters
    ----------
    A, B : array_like
           Points in space

    Returns
    -------
    a : array_like
    """
    AB = B - A
    a = AB / AB.dot(AB) ** 0.5
    return a
