Data loading task, add initial test cases