Fixing Notebook UI