started work on generate_labels.py