added missing comment and cleaned up a little