Added some update stuff