Added python impl - for demo reasons