#2. Add unit tests. Close #2.