Some hotfixes to atmo: Adding an import statement and an error in imaging