Remove update button and cleanup