Add a test for 'load' function