small refactor in tableview