Add initial tests
Make test passing with 100% coverage
Minor change to README