Working to make the update button show correctly.