Fixed an infinite loop!

And updated readme