got the layout to properly work