Python Code of mine