Add unit tests to Section