Adding more logic to page objects