Add decision tree and rearrange content
Clean up notebook, remove extra stuff