Removing main from Model