minor additional changes, sorry