Improved readme/comments