Ported the NitrogenAtom model