Improved README, improved code, added comments