Clean up 'load' function