Same as previous comment, It will be convenient to have logic part of code in page class itself and just call the methods from test case.