create first unit test + better Unit class