Update code.py

nn and amod transformed into 1 entity