(chore) First few unit tests!