improve layout

this can be removed