stack_boxes.py

def stack_boxes(level):

    #Find the sum of n numbers math formula --> n * (n+1) / 2
    sum_of_levels = (level * (level + 1))/2

    total_boxes = (2 * (sum_of_levels)) - level

    return f"Number of boxes for the level: {level} is: {int(total_boxes)}"

print(stack_boxes(4))
print(stack_boxes(2))
print(stack_boxes(0))
print(stack_boxes(1))
print(stack_boxes(10))
print(stack_boxes(100))