Seperated python module

Made changes to integrate better with intellij