Update .load() to return either an array of results or the single result