update comments and readme