fixing the layout