Simplify notebook sorting logic

[vikrantcue]

Currently the logic very complicated, have a simpler more readable function to do the sorting.