Add unit test section