Added boxes for visualization