Added python code, rearranged repo