Added buttons to screen + correct resizing of cell