fixed up the layout