Updated readme, fixed bug in Python implementation.