Make it work with Python