Add ami output.
DRY some