Added some comments and little corrections