Feat/update layouts