Turn into python package and restructure directories