Bug fix + update README