Add python module
Remove legacy stuff