need to fix main layout