fixed DB and adding more stuff to classify.py