cleaned up MLI code based on dry run feedback