Added in ability to remove boxes