Move unit tests to separate package