Add test coverage for added lines on main.