Completed unit tests for first two sections