Add unit test

This unit test is shitty but at least it works...