Added some comments and updated readme