changes for final screen to look good