Updated outputting and classified main.py