Implement heightForRowAtIndexPath to pass test.