Add unit test, fix errors
Get rid of extra println
Add some stuff to README