Added tests to check what's wrong