commented the code better and fixed the readme