Update code.py

Finished the short part.
Made all the results fit into one DataFrame
Was able to sort_index()