Refactored the way the load_all_tests code works