Add some comments, update readme