Debugging and cleaning code.
Fixed index bug.
Trying with more simple cases, just movement in x.
Playing with weights.