Python Code For Small Game