Improve README to better reflect the implementation