updated robust optimization

updated robust optimization  to work with changed optimization code