unit test help improve code quality