improve main layout