fixing layout

sorting out some issues