#13 added test assertions