Update weka.py

Edited. Works perfectly.