Improved README and comments