4.1 Just write unit-tests