Adding unit tests

right! Let's drop it ;)