Add correct layout