remove unused #includes that caused deprecation warnings
make sure to set the Model in the OptimizerState

close #465