Updated for new new model. Added --branch option for updating.