Iteration B2 - unit testing