woriking code with numpy