fix layout to look better