updated main.py to plot points live