updated layout on collapse button