Updated version of notebook with added documentation