Fixed some problems with update.