added more tests for load_argument_index and changed it so it passes