Update, fix bugs