try to make into a real python module