minor things to the code