Add first unit test
Saving work