adding unit tests, so committing