Adding missing os import