updated readme and fixed comments