Sprint #3: Unit tests. (Extra sprint)