Removed unnecessary comments and cleaned up code.
Added comments to clarify migration handling in load function.