Added one basic test
Unit testing