Added simple Unit tests