First working version. Still very ugly code.
Update README