Also fix the add more button