Added some comments, updated readme