Wrote first unit test (not tested yet)