Custom env as optional argument for the atom decorator