Add VC test and code for heightForRowAtIndexPath.