Looping fixes. Fixes #2