Add more unit tests and comment code