Some unit tests for the new features