Initial commit for visualize_box method