Test

Add a TODO here to add more test. I am not sure how we'll add more tests here, but should figure out a mechanism in the future to do that