minor changes, trying to fix the layout