[WIP] Test into

Changed one function around. Wasnt working before, fixed it, added chunksize argument. Not sure if we want this to be a pattern, where we add chunksize arguments for functions which use chunks.

Also the code looks ugly and can probably be done better. 
maybe make this `range(4)` and `chunksize=2` below