Fix, still dirty. Need to learn Python (XD)