small changes to layout again