Break box component into its own class