updated for new notebook