ADD: vector_square function