Add container styles for simple visualization