Updated code for testing the k truck cases