Improve imports statement
Add support for optimx package