Small changes to the layout