Add some test and refactoring code