adding Nick D's hw

need to import os for this to work