Added tests (#1)