Adding unit tests, updating README