Move updating containers to reload() function

This avoids calling Register() multiple times which isn't working.
This fixes #1.