Optimizing Box Weights Hackerrank Practice