Change layout to make style work