Refine code with unittest and page object model(1)