refacot methods
modify test case