Adding some comments

Some comments were added, to make it more readable