minor changes

updating comments and some parts of the code