identified that shape an array takes a lot of time so you need to get back to the old method

use

```
def function(u,w,v,t)
```

instead of 

```
def function(x,t)
```

and split x in the function.

Because reshape a function can take a long time.
