Added python implementation

Import this python implementation YO