Added tests to the code