Update code.py

capital letters fixes, issue with AMs solved