Improved doc comments, updated readme