Basic vector operations with numpy