Add the collapse button