Update main.py

works up to writing notes in a directory.