Add python code and moved folders around