Adding an update script...still buggy