Add some comments and modify readme