Import os correctly