Code For Small Game In Python