changed some stuff, made a train.py so we can actually train our bots