Pull python code to top level and remove other things