small fixes after comments