Add colapse div function