Split up notebook into two notebooks
One notebook concerning code (model helper)
The other notebook concerning declarations (early vision model)
Move flatten_tuple to utility functions