Small fixes to profiling notebook