doesn't work if mulitple time steps due to distance scipy array