Added some more comments and updated readme