Added remove box function, untested