added working python prediction script
fixed a buncha wrong shit