Update README.md

Making code compilable ;)