Add method argument to robust fit method