New python codes

Yeeeeee