Bug fixes. Update README