Adding unit tests to latest features