add the os module and test code