#5 Added tests.