Making this into a python project