Add container to make it look nicer on narrow devices