Corrections for better nbviewer visualization