Beginning of unit_vector exercise