modification after test on numergy