LOL
Add unit tests