Simple test case
Fixed iterating over all possible weights, test case passed