Fix bug + readme