Add unit vector transformation - Fix bug