Add os import to notebook, accidentally removed