add some comments and fix readme