Added assert. Added tests.