slight modifications to the code

not many but only a little