Added ability to draw bounding box