Update README
Adding unit tests & making tests pass