Fixed plot output in notebook